Deep learning framework for prediction of infection severity of COVID-19.

With the onset of the COVID-19 pandemic, quantifying the condition of positively diagnosed patients is of paramount importance. Chest CT scans can be used to measure the severity of a lung infection and the isolate involvement sites in order to increase awareness of a patient's disease progression. In this work, we developed a deep learning framework for lung infection severity prediction. To this end, we collected a dataset of 232 chest CT scans and involved two public datasets with an additional 59 scans for our model's training and used two external test sets with 21 scans for evaluation. On an input chest Computer Tomography (CT) scan, our framework, in parallel, performs a lung lobe segmentation utilizing a pre-trained model and infection segmentation using three distinct trained SE-ResNet18 based U-Net models, one for each of the axial, coronal, and sagittal views. By having the lobe and infection segmentation masks, we calculate the infection severity percentage in each lobe and classify that percentage into 6 categories of infection severity score using a k-nearest neighbors (k-NN) model. The lobe segmentation model achieved a Dice Similarity Score (DSC) in the range of [0.918, 0.981] for different lung lobes and our infection segmentation models gained DSC scores of 0.7254 and 0.7105 on our two test sets, respectfully. Similarly, two resident radiologists were assigned the same infection segmentation tasks, for which they obtained a DSC score of 0.7281 and 0.6693 on the two test sets. At last, performance on infection severity score over the entire test datasets was calculated, for which the framework's resulted in a Mean Absolute Error (MAE) of 0.505 ± 0.029, while the resident radiologists' was 0.571 ± 0.039.

Statistical analysis of COVID-19 infection severity in lung lobes from chest CT.

Detection of the COVID 19 virus is possible through the reverse transcription-polymerase chain reaction (RT-PCR) kits and computed tomography (CT) images of the lungs. Diagnosis via CT images provides a faster diagnosis than the RT-PCR method does. In addition to low false-negative rate, CT is also used for prognosis in determining the severity of the disease and the proposed treatment method. In this study, we estimated a probability density function (PDF) to examine the infections caused by the virus. We collected 232 chest CT of suspected patients and had them labeled by two radiologists in 6 classes, including a healthy class and 5 classes of different infection severity. To segment the lung lobes, we used a pre-trained U-Net model with an average Dice similarity coefficient (DSC) greater than 0.96. First, we extracted the PDF to grade the infection of each lobe and selected five specific thresholds as feature vectors. We then assigned this feature vector to a support vector machine (SVM) model and made the final prediction of the infection severity. Using the T-Test statistics, we calculated the p-value at different pixel thresholds and reported the significant differences in the pixel values. In most cases, the p-value was less than 0.05. Our developed model was developed on roughly labeled data without any manual segmentation, which estimated lung infection involvements with the area under the curve (AUC) in the range of [0.64, 0.87]. The introduced model can be used to generate a systematic automated report for individual patients infected by COVID-19.